naive bayes python code from scratch